# -*- coding: utf-8 -*-
# ===========================================
# @Time    : 2021/11/24 12:06 
# @Author  : shutao
# @FileName: hand_key_point.py
# @remark  : 
# 
# @Software: PyCharm
# Github 　： https://github.com/NameLacker
# ===========================================

import numpy as np
import paddle

from .base_dataset import BaseDataset

class HandKeyPointDataset(BaseDataset):
    def __init__(self, root_path, is_test=False):
        super(HandKeyPointDataset, self).__init__()
        self.numbers = 100
        self.img_shape = (3, 768, 768)

    def __getitem__(self, item):
        data = paddle.uniform(self.img_shape)
        label = paddle.randint(0, 10, [1])

        return data, label

    def __len__(self):
        return self.numbers
